# 开始游戏项目练习《外星人入侵》

# 按 使用左右箭头左右移动飞船，还可以用空格射击
# 这个是文件的主运行文件


import sys
import pygame
from settings import Settings
from ship import Ship
from bullet import Bullet
from alien import Alien


class AlienInvasion:
    """管理游戏资源和行为的类"""

    def __init__(self):
        """初始化游戏并创建游戏资源"""
        pygame.init()
        self.settings = Settings()
        self.screen = pygame.display.set_mode((0, 0), pygame.FULLSCREEN)  # 设置全屏框
        self.settings.screen_width = self.screen.get_rect().width
        self.settings.screen_height = self.screen.get_rect().height
        self.settings.screen_left = self.screen.get_rect().left

        pygame.display.set_caption("Alien Invasion")  # 边框标题

        self.ship = Ship(self)  # 创建一艘飞船
        self.bullets = pygame.sprite.Group()  # 创建用于存储子弹的编组
        self.aliens = pygame.sprite.Group()  # 创建用于存储外星人的编组

        self._create_fleet()  # 编写创建一群外星人的方法

        # 设置背景颜色
        self.bg_color = (230, 230, 230)

    def run_game(self):
        """开始游戏的主循环"""
        while True:
            self._check_events()  # 监听鼠标事件
            self.ship.update()  # 更新飞船位置的事件
            self._update_bullets()  # 子弹更新事件
            self._update_aliens()  # 外星人更新事件
            self._update_screen()  # 更新屏幕画布

    # 飞船和子弹函数
    def _check_events(self):
        # 监视键盘和鼠标事件
        for event in pygame.event.get():
            if event.type == pygame.QUIT:  # 退出
                sys.exit()
            elif event.type == pygame.KEYDOWN:  # 按下操作
                self._check_keydown_events(event)
            elif event.type == pygame.KEYUP:  # 抬起操作
                self._check_keyup_events(event)

    def _check_keydown_events(self, event):  # 优化结构产物，该函数为鼠标按下操作
        if event.key == pygame.K_RIGHT:  # 向右移动飞船
            self.ship.moving_right = True
        elif event.key == pygame.K_LEFT:  # 向左移动飞船
            self.ship.moving_left = True
        elif event.key == pygame.K_q:  # 按q退出游戏
            sys.exit()
        elif event.key == pygame.K_SPACE:  # 按空格发射子弹
            self._fire_bullet()

    def _check_keyup_events(self, event):  # 优化结构产物，该函数为鼠标抬起操作
        if event.key == pygame.K_RIGHT:  # 鼠标抬起，更改移动标识
            self.ship.moving_right = False
        elif event.key == pygame.K_LEFT:
            self.ship.moving_left = False

    def _update_bullets(self):
        self.bullets.update()  # 更新子弹位置
        # 删除消失的子弹
        for bullet in self.bullets.copy():
            if bullet.rect.bottom <= 0:
                self.bullets.remove(bullet)
        # 检查是否有子弹击中外星人，如果是就删除相应的子弹和外星人
        collisions = pygame.sprite.groupcollide(self.bullets, self.aliens, True, True)

    def _fire_bullet(self):
        """创建一颗子弹，并将其加入到编组bullets中"""
        if len(self.bullets) < self.settings.bullet_allowed:
            new_bullet = Bullet(self)
            self.bullets.add(new_bullet)

    # 外星人函数
    def _create_fleet(self):
        """创建外星人群"""
        # # 创建第一个外星人
        # alien = Alien(self)
        # self.aliens.add(alien)
        # 创建一个外星人并计算一行可以容纳多少外星人
        alien = Alien(self)
        alien_width, alien_height = alien.rect.size  # 获取外星人的宽度和高度
        available_space_x = self.settings.screen_width - (2 * alien_width)  # 获取可以放外星人的长度
        number_aliens_x = available_space_x // (2 * alien_width)  # 计算得到表示外星人个数的整数

        # 计算屏幕可以容纳多少行外星人
        ship_height = self.ship.rect.height
        available_space_y = (self.settings.screen_height - (3 * alien_height) - ship_height)  # 计算可用的外星人的y长度
        number_rows = available_space_y // (2 * alien_height)

        # 创建外星人群
        for row_number in range(number_rows):
            for alien_number in range(number_aliens_x):
                self._create_alien(alien_number, row_number)

    def _create_alien(self, alien_number, row_number):
        # 创建第一个外星人并将其加入当前行
        alien = Alien(self)
        alien_width, alien_height = alien.rect.size
        alien.x = alien_width + 2 * alien_width * alien_number  # 设置外星人x的坐标
        alien.rect.x = alien.x
        alien.rect.y = alien.rect.height + 2 * alien.rect.height * row_number
        self.aliens.add(alien)

    def _update_aliens(self):
        """更新外星人群中所有外星人的位置"""
        self._check_fleet_edges()  # 检查是否有外星人位于边缘
        self.aliens.update()

    def _check_fleet_edges(self):
        """有外星人到达边缘时采用的相应的措施"""
        for alien in self.aliens.sprites():
            if alien.check_edges():
                self._change_fleet_direction()
                break

    def _change_fleet_direction(self):
        """将整群外星人下一，并改变他们的方向"""
        for alien in self.aliens.sprites():
            alien.rect.y += self.settings.fleet_drop_speed
        self.settings.fleet_direction *= -1

    # 屏幕绘制
    def _update_screen(self):
        # 每次循环时都重新绘制屏幕
        self.screen.fill(self.settings.bg_color)  # 添加窗口背景颜色
        self.ship.blitme(self)  # 将飞船绘制在屏幕上
        for bullet in self.bullets.sprites():
            bullet.draw_bullet()
        self.aliens.draw(self.screen)  # 将外星人绘制在屏幕上
        # 让最近绘制的屏幕可见
        pygame.display.flip()


if __name__ == '__main__':
    # 创建游戏实例并运行游戏。
    ai = AlienInvasion()
    ai.run_game()
